package org.common.utils.fasttest.util.enums;
import org.eclipse.collections.impl.map.mutable.ConcurrentHashMap;
import java.lang.reflect.Field;
import java.util.Map;

/**
 * 枚举工具
 * @author zhouzhibing
 * @date 2024/11/27
 */
public class EnumUtil {

    /**
     * 枚举实体列表<类型，<枚举ID/key，枚举对象>
     */
    private static final Map<Class<?> , Map<Integer, Object>> ENUM_INSTANCE_MAP = new ConcurrentHashMap<>();

    /**
     * 根据指定类型枚举和key，返回对应的枚举对象。
     * @param clzss 指定类型枚举
     * @param id 指定枚举中的key id
     * @return 返回枚举对象
     * @param <T> 枚举泛型类型
     */
    public static <T> T of(Class<T> clzss , int id) {
        Map<Integer, Object> enumMap = ENUM_INSTANCE_MAP.get(clzss);
        if(enumMap != null)
            return (T)enumMap.get(id);

        init(clzss);
        return of(clzss , id);
    }

    /**
     * 初始化指定类型枚举
     * @param clzss 指定类型枚举
     * @param <T> 枚举泛型类型
     */
    private static <T> void init(Class<T> clzss) {
        Field fieldKey = getFieldKey(clzss);
        try {
            fieldKey.setAccessible(true);
            T[] enumConstants = clzss.getEnumConstants();
            for (T enumInstance : enumConstants) {
                int key = fieldKey.getInt(enumInstance);
                Map<Integer, Object> enumMap = ENUM_INSTANCE_MAP.computeIfAbsent(clzss, (k) -> new ConcurrentHashMap<>());
                enumMap.put(key , enumInstance);
            }
        } catch (IllegalAccessException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 根据类型获取指定EnumKey的字段对象
     * @param clzss  类型
     * @return 获取指定EnumKey的字段对象
     * @param <T> 枚举泛型类型
     */
    private static <T> Field getFieldKey(Class<T> clzss) {
        Field fieldKey = null;
        Field[] fields = clzss.getDeclaredFields();
        for (Field field : fields) {
            EnumKey enumKey = field.getAnnotation(EnumKey.class);
            if(enumKey != null) {
                fieldKey = field;
                break;
            }
        }

        //没有枚举key
        if(fieldKey == null) {
            throw new RuntimeException(clzss.getName() + ", fieldKey == null, not exist @EnumKey.");
        }

        //不是该数值类型
        if (fieldKey.getType() != int.class && fieldKey.getType() != Integer.class) {
            throw new RuntimeException(clzss.getName() + "." + fieldKey.getName() + ", field type not int or Integer .");
        }
        return fieldKey;
    }
}
